package com.boot.fileupload.controller;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

@Controller
public class UpLoadController {


    @GetMapping("/")
    public ModelAndView index() {
        ModelAndView modelAndView = new ModelAndView("/index/upload");

        return modelAndView;
    }

    @Value("C:\\Users\\Administrator\\Desktop\\")
    private String UPLOADED_FOLDER;

    @PostMapping("/upload")
    public String singleFileUpload(
            @RequestParam("file") MultipartFile file,
            RedirectAttributes redirectAttributes
                                  ) {

        /**
         * 重定向的属性值,如果重定向,response将不能用, 所以用redirect属性
         */
        if (file.isEmpty()) {
            redirectAttributes.addFlashAttribute("message", "Please select a file to upload");
            return "redirect:uploadStatus";
        }

        try {
            // Get the file and save it somewhere
            byte[] bytes = file.getBytes();
            Path path = Paths.get(UPLOADED_FOLDER + file.getOriginalFilename());
            Files.write(path, bytes);

            redirectAttributes.addFlashAttribute(
                    "message",
                    "You successfully uploaded '" + file.getOriginalFilename() + "'"
                                                );

        } catch (IOException e) {
            e.printStackTrace();
        }

        return "redirect:/index/uploadStatus";
    }

}
